package com.tensquare.base.service;

import com.tensquare.base.dao.LabelDao;
import com.tensquare.base.pojo.Label;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Pageable;
import org.springframework.data.jpa.domain.Specification;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import util.IdWorker;

import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root;
import java.util.ArrayList;
import java.util.List;


@Service
@Transactional
public class LabelService {

    @Autowired
    private LabelDao labelDao;

    @Autowired
    private IdWorker idWorker;

    /**
     * 查询所有
     */
    public List<Label> findAll() {
        return labelDao.findAll();
    }

    /**
     * 根据id查询
     */
    public Label findById(String id) {
        return labelDao.findById(id).get();
    }

    /**
     * 增加标签
     */
    public void save(Label label) {
        label.setId(idWorker.nextId() + "");
        labelDao.save(label);
    }

    /**
     * 修改
     */
    public void update(Label label) {
        labelDao.save(label);
    }

    /**
     * 根据id删除
     */
    public void deleteById(String id) {
        labelDao.deleteById(id);
    }

    /**
     * 标签分页
     */
    public List<Label> findSearch(Label label) {
        // 根据条件查询
        return labelDao.findAll(new Specification<Label>() {
            /**
             * 返回null，表示不需要任何条件
             * @param root  根对象，也就是要把条件封装到哪个对象中，相当于 where 列名=label.getId
             *              （通过root用“类的属性名”来获取“列名”）
             * @param query 封装的都是查询关键字，比如group by、order by
             * @param cb    用来封装条件对象（代表查询条件，调用不同的方法 表示设置了 不同的查询条件）
             *              参数：列名，列的值
             * @return
             */
            @Override
            public Predicate toPredicate(Root<Label> root, CriteriaQuery<?> query, CriteriaBuilder cb) {
                // 创建一个List，用来存放所有条件
                List<Predicate> list = new ArrayList<>();
                if (null != label.getLabelname() && "" != label.getLabelname()) {
                    // 设置条件：where labelname like %name%
                    Predicate predicate = cb.like(root.get("labelname").as(String.class), "%" + label.getLabelname() + "%");
                    list.add(predicate);
                }

                if (null != label.getState() && !"".equals(label.getState())){
                    Predicate predicate = cb.equal(root.get("state").as(String.class), label.getState());
                    list.add(predicate);
                }

                Predicate[] parr = new Predicate[list.size()];
                list.toArray(parr);
                // cb.and()是把所有的添加都组合起来
                return cb.and(parr);
            }
        });
    }

    /**
     * 分页查询
     * @param label
     * @return
     */
    public Page<Label> pageQuery(Label label, int page, int size) {

        // 设置分页参数
        Pageable pageable = PageRequest.of(page-1, size);

        // 设置条件
        Specification<Label> specification = new Specification<Label>() {
            @Override
            public Predicate toPredicate(Root<Label> root, CriteriaQuery<?> criteriaQuery, CriteriaBuilder cb) {
                List<Predicate> list = new ArrayList<>();

                if(null != label.getLabelname() && !"".equals(label.getLabelname().trim())){
                    Predicate predicate = cb.like(root.get("labelname").as(String.class), "%" + label.getLabelname() + "%");
                    list.add(predicate);
                }

                if(null != label.getState() && !"".equals(label.getLabelname().trim())){
                    Predicate predicate = cb.equal(root.get("").as(String.class), "%" + label.getState() + "%");
                    list.add(predicate);
                }

                Predicate[] parr = new Predicate[list.size()];
                list.toArray(parr);

                return cb.and(parr);
            }
        };

        return labelDao.findAll(specification, pageable);
    }
}
